package woa.array;

/**
 * 数组中的第K个最大元素
 * 给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。
 * 请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。
 * 输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
 * 输出: 4
 *
 * @author wangpeng
 * @date 2021/11/10
 */
public class FindKthLargestProblem {

    /**
     * 快排思路、分治思想、递推编码
     *
     * @param nums
     * @param k
     * @return
     */
    public static int findKthLargest(int[] nums, int k) {
        if (nums.length == 1) {
            return nums[0];
        }
        int r = find(nums, k, 0, nums.length - 1);
        return nums[r];
    }

    public static int find(int[] nums, int k, int p, int q) {
//        if (p >= q) {
//            return p;
//        }
        int r = partition(nums, p, q);
        if (r + 1 == k) {
            return r;
        }
        if (r + 1 > k) {
            return find(nums, k, p, r - 1);
        } else {
            return find(nums, k, r, q);
        }
    }

    public static int partition(int[] nums, int p, int q) {
        int i = p;
        for (int j = p; j < q; j++) {
            if (nums[j] >= nums[q]) {
                int temp = nums[j];
                nums[j] = nums[i];
                nums[i] = temp;
                i++;
            }
        }
        int temp = nums[i];
        nums[i] = nums[q];
        nums[q] = temp;

        return i;
    }

    public static void main(String[] args) {
        System.out.println(findKthLargest(new int[]{2,2}, 2));
    }
}
